home *** CD-ROM | disk | FTP | other *** search
/ Amiga Plus 2002 #11 / Amiga Plus CD - 2002 - No. 11.iso / Tools / Development / libogg / libvorbis-1.0rc3 / vq / huffbuild.c < prev    next >
Encoding:
C/C++ Source or Header  |  2002-10-27  |  5.9 KB  |  223 lines
  1. /********************************************************************
  2.  *                                                                  *
  3.  * THIS FILE IS PART OF THE OggVorbis SOFTWARE CODEC SOURCE CODE.   *
  4.  * USE, DISTRIBUTION AND REPRODUCTION OF THIS LIBRARY SOURCE IS     *
  5.  * GOVERNED BY A BSD-STYLE SOURCE LICENSE INCLUDED WITH THIS SOURCE *
  6.  * IN 'COPYING'. PLEASE READ THESE TERMS BEFORE DISTRIBUTING.       *
  7.  *                                                                  *
  8.  * THE OggVorbis SOURCE CODE IS (C) COPYRIGHT 1994-2001             *
  9.  * by the XIPHOPHORUS Company http://www.xiph.org/                  *
  10.  *                                                                  *
  11.  ********************************************************************
  12.  
  13.  function: hufftree builder
  14.  last mod: $Id: huffbuild.c,v 1.12 2001/12/20 01:00:39 segher Exp $
  15.  
  16.  ********************************************************************/
  17.  
  18. #include <stdlib.h>
  19. #include <string.h>
  20. #include <math.h>
  21. #include <stdio.h>
  22. #include "bookutil.h"
  23.  
  24. static int nsofar=0;
  25. static int getval(FILE *in,int begin,int n,int group,int max){
  26.   float v;
  27.   int i;
  28.   long val=0;
  29.  
  30.   if(nsofar>=n || get_line_value(in,&v)){
  31.     reset_next_value();
  32.     nsofar=0;
  33.     if(get_next_value(in,&v))
  34.       return(-1);
  35.     for(i=1;i<=begin;i++)
  36.       get_line_value(in,&v);
  37.   }
  38.  
  39.   val=(int)v;
  40.   nsofar++;
  41.  
  42.   for(i=1;i<group;i++,nsofar++)
  43.     if(nsofar>=n || get_line_value(in,&v))
  44.       return(getval(in,begin,n,group,max));
  45.     else
  46.       val = val*max+(int)v;
  47.   return(val);
  48. }
  49.  
  50. static void usage(){
  51.   fprintf(stderr,
  52.       "usage:\n" 
  53.       "huffbuild <input>.vqd <begin,n,group>|<lorange-hirange> [noguard]\n"
  54.       "   where begin,n,group is first scalar, \n"
  55.       "                          number of scalars of each in line,\n"
  56.       "                          number of scalars in a group\n"
  57.       "eg: huffbuild reslongaux.vqd 0,1024,4\n"
  58.       "produces reslongaux.vqh\n\n");
  59.   exit(1);
  60. }
  61.  
  62. int main(int argc, char *argv[]){
  63.   char *base;
  64.   char *infile;
  65.   int i,j,k,begin,n,subn,guard=1;
  66.   FILE *file;
  67.   int maxval=0;
  68.   int loval=0;
  69.  
  70.   if(argc<3)usage();
  71.   if(argc==4)guard=0;
  72.  
  73.   infile=strdup(argv[1]);
  74.   base=strdup(infile);
  75.   if(strrchr(base,'.'))
  76.     strrchr(base,'.')[0]='\0';
  77.  
  78.   {
  79.     char *pos=strchr(argv[2],',');
  80.     char *dpos=strchr(argv[2],'-');
  81.     if(dpos){
  82.       loval=atoi(argv[2]);
  83.       maxval=atoi(dpos+1);
  84.       subn=1;
  85.       begin=0;
  86.     }else{
  87.       begin=atoi(argv[2]);
  88.       if(!pos)
  89.     usage();
  90.       else
  91.     n=atoi(pos+1);
  92.       pos=strchr(pos+1,',');
  93.       if(!pos)
  94.     usage();
  95.       else
  96.     subn=atoi(pos+1);
  97.       if(n/subn*subn != n){
  98.     fprintf(stderr,"n must be divisible by group\n");
  99.     exit(1);
  100.       }
  101.     }
  102.   }
  103.  
  104.   /* scan the file for maximum value */
  105.   file=fopen(infile,"r");
  106.   if(!file){
  107.     fprintf(stderr,"Could not open file %s\n",infile);
  108.     if(!maxval)
  109.       exit(1);
  110.     else
  111.       fprintf(stderr,"  making untrained books.\n");
  112.  
  113.   }
  114.  
  115.   if(!maxval){
  116.     i=0;
  117.     while(1){
  118.       long v;
  119.       if(get_next_ivalue(file,&v))break;
  120.       if(v>maxval)maxval=v;
  121.       
  122.       if(!(i++&0xff))spinnit("loading... ",i);
  123.     }
  124.     rewind(file);
  125.     maxval++;
  126.   }
  127.  
  128.   {
  129.     long vals=pow(maxval,subn);
  130.     long *hist=_ogg_malloc(vals*sizeof(long));
  131.     long *lengths=_ogg_malloc(vals*sizeof(long));
  132.     
  133.     for(j=loval;j<vals;j++)hist[j]=guard;
  134.     
  135.     if(file){
  136.       reset_next_value();
  137.       i/=subn;
  138.       while(!feof(file)){
  139.     long val=getval(file,begin,n,subn,maxval);
  140.     if(val==-1 || val>=vals)break;
  141.     hist[val]++;
  142.     if(!(i--&0xff))spinnit("loading... ",i*subn);
  143.       }
  144.       fclose(file);
  145.     }
  146.  
  147.     /* we have the probabilities, build the tree */
  148.     fprintf(stderr,"Building tree for %ld entries\n",vals);
  149.     build_tree_from_lengths0(vals,hist,lengths);
  150.  
  151.     /* save the book */
  152.     {
  153.       char *buffer=alloca(strlen(base)+5);
  154.       strcpy(buffer,base);
  155.       strcat(buffer,".vqh");
  156.       file=fopen(buffer,"w");
  157.       if(!file){
  158.     fprintf(stderr,"Could not open file %s\n",buffer);
  159.     exit(1);
  160.       }
  161.     }
  162.     
  163.     fprintf(file,
  164.   "/********************************************************************\n"
  165.   " *                                                                  *\n"
  166.   " * THIS FILE IS PART OF THE OggVorbis SOFTWARE CODEC SOURCE CODE.   *\n"
  167.   " * USE, DISTRIBUTION AND REPRODUCTION OF THIS LIBRARY SOURCE IS     *\n"
  168.   " * GOVERNED BY A BSD-STYLE SOURCE LICENSE INCLUDED WITH THIS SOURCE *\n"
  169.   " * IN 'COPYING'. PLEASE READ THESE TERMS BEFORE DISTRIBUTING.       *\n"
  170.   " *                                                                  *\n"
  171.   " * THE OggVorbis SOURCE CODE IS (C) COPYRIGHT 1994-2001             *\n"
  172.   " * by the XIPHOPHORUS Company http://www.xiph.org/                  *\n"
  173.   " *                                                                  *\n"
  174.   " ********************************************************************\n"
  175.   "\n"
  176.   " function: static codebook autogenerated by huff/huffbuld\n"
  177.   "\n"
  178.   " ********************************************************************/\n\n");
  179.  
  180.     fprintf(file,"#ifndef _V_%s_VQH_\n#define _V_%s_VQH_\n",base,base);
  181.     fprintf(file,"#include \"codebook.h\"\n\n");
  182.     
  183.     /* first, the static vectors, then the book structure to tie it together. */
  184.     /* lengthlist */
  185.     fprintf(file,"static long _huff_lengthlist_%s[] = {\n",base);
  186.     for(j=0;j<vals;){
  187.       fprintf(file,"\t");
  188.       for(k=0;k<16 && j<vals;k++,j++)
  189.     fprintf(file,"%2ld,",lengths[j]);
  190.       fprintf(file,"\n");
  191.     }
  192.     fprintf(file,"};\n\n");
  193.     
  194.     /* the toplevel book */
  195.     fprintf(file,"static static_codebook _huff_book_%s = {\n",base);
  196.     fprintf(file,"\t%d, %ld,\n",subn,vals);
  197.     fprintf(file,"\t_huff_lengthlist_%s,\n",base);
  198.     fprintf(file,"\t0, 0, 0, 0, 0,\n");
  199.     fprintf(file,"\tNULL,\n");
  200.  
  201.     fprintf(file,"\tNULL,\n");
  202.     fprintf(file,"\tNULL,\n");
  203.     fprintf(file,"\tNULL,\n");
  204.     fprintf(file,"\t0\n};\n\n");
  205.     
  206.     fprintf(file,"\n#endif\n");
  207.     fclose(file);
  208.     fprintf(stderr,"Done.                                \n\n");
  209.   }
  210.   exit(0);
  211. }
  212.  
  213.  
  214.  
  215.  
  216.  
  217.  
  218.  
  219.  
  220.  
  221.  
  222.  
  223.